﻿using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using NPOI;
using NPOI.HPSF;
using NPOI.HSSF;
using NPOI.HSSF.UserModel;
using NPOI.POIFS;
using NPOI.Util;
using NPOI.XSSF.UserModel;
using NPOI.SS.UserModel;
using NPOI.POIFS.FileSystem;
using System.IO;
using System.Data;

namespace ExcelUtli
{
    public class ExcelOpr
    {
        /// <summary>
        /// Read Excel To DataTable
        /// </summary>
        /// <param name="fileName"></param>
        /// <param name="sheetName"></param>
        /// <param name="isFirstRowColumn"></param>
        /// <returns></returns>
        public static DataTable ReadToTableCheckDateTime(String fileName, string sheetName = null, bool isFirstRowColumn = true)
        {
            DataTable data = new DataTable();
            NPOI.SS.UserModel.ISheet sheet = null;
            int startRow = 0;
            try
            {
                if (!File.Exists(fileName))
                {
                    return null;
                }
                FileInfo fileInfo = new FileInfo(fileName);
                IWorkbook workbook = null;
                //MemoryStream m = new MemoryStream();
                FileStream fs = new FileStream(fileName, FileMode.Open, FileAccess.Read);
                //POIFSFileSystem POfs = new POIFSFileSystem(fs);
                if (fileInfo.Extension.ToLower().Equals(".xls"))
                {
                    workbook = new HSSFWorkbook(fs);

                }
                else
                {
                    // NPOI.SS.UserModel.IWorkbook workbook = NPOI.SS.UserModel.WorkbookFactory.Create(fs);
                    workbook = NPOI.SS.UserModel.WorkbookFactory.Create(fs);
                }


                if (!string.IsNullOrEmpty(sheetName))
                {
                    sheet = workbook.GetSheet(sheetName);
                    if (sheet == null)
                    {
                        sheet = workbook.GetSheetAt(0);
                    }
                }
                else
                {
                    sheet = workbook.GetSheetAt(0);
                }
                if (sheet != null)
                {
                    NPOI.SS.UserModel.IRow firstRow = sheet.GetRow(0);
                    int cellCount = firstRow.LastCellNum;
                    if (isFirstRowColumn)
                    {
                        for (int i = firstRow.FirstCellNum; i < cellCount; ++i)
                        {
                            ICell cell = firstRow.GetCell(i);
                            if (cell != null)
                            {
                                string cellValue = cell.StringCellValue;
                                if (cellValue != null)
                                {
                                    DataColumn column = new DataColumn(cellValue);
                                    data.Columns.Add(column);
                                }
                            }
                        }
                        startRow = sheet.FirstRowNum + 1;
                    }
                    else
                    {
                        startRow = sheet.FirstRowNum;
                    }
                    //最后一列的标号
                    int rowCount = sheet.LastRowNum;
                    for (int i = startRow; i <= rowCount; ++i)
                    {
                        IRow row = sheet.GetRow(i);
                        if (row == null || row.FirstCellNum < 0)
                            continue;
                        DataRow dataRow = data.NewRow();
                        for (int j = row.FirstCellNum; j < cellCount; ++j)
                        {
                            //没有数据的单元格都默认是Null
                            ICell cell = row.GetCell(j);
                            if (cell != null)
                            {
                                if (cell.CellType == CellType.Numeric)
                                {
                                    //                    if (cell.getCellStyle().getDataFormatString() cell.CellType.GetType().GetTypeCode()== 58) {  
                                    //// 处理自定义日期格式：m月d日(通过判断单元格的格式id解决，id的值是58)  
                                    //SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");  
                                    //double value = cell.getNumericCellValue();  
                                    //Date date = org.apache.poi.ss.usermodel.DateUtil  
                                    //        .getJavaDate(value);  
                                    //result = sdf.format(date);  
                                    //short format = cell.CellStyle.DataFormat;
                                    //if (format == 31)
                                    //{
                                    //    DateTime date = cell.DateCellValue;
                                    //    dataRow[j] = date.ToString("yyy-MM-dd");
                                    //}
                                    //else 
                                    if (DateUtil.IsCellDateFormatted(cell))
                                    {
                                        dataRow[j] = row.GetCell(j).DateCellValue;

                                    }
                                    else
                                    {
                                        dataRow[j] = row.GetCell(j).ToString().Trim();
                                    }
                                }
                                else
                                {
                                    dataRow[j] = row.GetCell(j).ToString().Trim();
                                }
                            }
                        }
                        data.Rows.Add(dataRow);
                    }

                }
                return data;
            }
            catch (Exception ex)
            {
                LogWrite.WriteLog(string.Format("加载出错,原因{0}", ex.ToString()));
                return null;
            }
        }
        /// <summary>
        /// 将DataTable数据导入到excel中
        /// </summary>
        /// <param name="data">要导入的数据</param>
        /// <param name="isColumnWritten">DataTable的列名是否要导入</param>
        /// <param name="sheetName">要导入的excel的sheet的名称</param>
        /// <returns>导入数据行数(包含列名那一行)</returns>
        public static int DataTableToExcel(DataTable data, string fileName,string sheetName="ParamSheet", bool isColumnWritten=true)
        {
            int i = 0;
            int j = 0;
            int count = 0;
            ISheet sheet = null;
            IWorkbook workbook = null;
            FileStream fs = new FileStream(fileName, FileMode.OpenOrCreate, FileAccess.ReadWrite);
            if (fileName.IndexOf(".xlsx") > 0) // 2007版本
                workbook = new XSSFWorkbook();
            else if (fileName.IndexOf(".xls") > 0) // 2003版本
                workbook = new HSSFWorkbook();
            try
            {
                if (workbook != null)
                {
                    sheet = workbook.CreateSheet(sheetName);
                }
                else
                {
                    return -1;
                }

                if (isColumnWritten == true) //写入DataTable的列名
                {
                    IRow row = sheet.CreateRow(0);
                    for (j = 0; j < data.Columns.Count; ++j)
                    {
                        row.CreateCell(j).SetCellValue(data.Columns[j].ColumnName);
                    }
                    count = 1;
                }
                else
                {
                    count = 0;
                }

                for (i = 0; i < data.Rows.Count; ++i)
                {
                    IRow row = sheet.CreateRow(count);
                    for (j = 0; j < data.Columns.Count; ++j)
                    {
                        if (data.Rows[i][j] is string)
                            row.CreateCell(j).SetCellValue(data.Rows[i][j].ToString());
                        else
                            row.CreateCell(j).SetCellValue("");
                    }
                    ++count;
                }
                workbook.Write(fs); //写入到excel
                return count;
            }
            catch (Exception ex)
            {
                LogWrite.WriteLog("导出错误，原因 " + ex.Message);
                return -1;
            }
        }
    }
}
